package com.coursehub.progress.repository;

import com.coursehub.progress.entity.StudentExercise;
import com.coursehub.progress.entity.StudentExerciseId;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.List;
import java.util.Optional;

/**
 * 学生练习完成记录Repository
 */
@Repository
public interface StudentExerciseRepository extends JpaRepository<StudentExercise, StudentExerciseId> {
    
    /**
     * 根据用户ID查找所有已完成的练习
     */
    List<StudentExercise> findByUserIdAndIsFinishedTrue(String userId);
    
    /**
     * 根据用户ID查找所有未完成的练习
     */
    List<StudentExercise> findByUserIdAndIsFinishedFalse(String userId);
    
    /**
     * 根据用户ID和练习ID查找记录
     */
    Optional<StudentExercise> findByUserIdAndExerciseId(String userId, String exerciseId);
    
    /**
     * 根据练习ID查找所有完成记录
     */
    List<StudentExercise> findByExerciseIdAndIsFinishedTrue(String exerciseId);
    
    /**
     * 根据用户ID查找已完成的练习（按练习ID列表过滤）
     */
    @Query("SELECT se FROM StudentExercise se WHERE se.userId = :userId AND se.isFinished = true AND se.exerciseId IN :exerciseIds")
    List<StudentExercise> findCompletedExercisesByUserAndExerciseIds(@Param("userId") String userId, @Param("exerciseIds") List<String> exerciseIds);
    
    /**
     * 根据用户ID查找未完成的练习（按练习ID列表过滤）
     */
    @Query("SELECT se FROM StudentExercise se WHERE se.userId = :userId AND se.isFinished = false AND se.exerciseId IN :exerciseIds")
    List<StudentExercise> findUnfinishedExercisesByUserAndExerciseIds(@Param("userId") String userId, @Param("exerciseIds") List<String> exerciseIds);
    
    /**
     * 检查用户是否完成了指定练习
     */
    boolean existsByUserIdAndExerciseIdAndIsFinishedTrue(String userId, String exerciseId);
    
    /**
     * 统计用户完成的练习数量
     */
    long countByUserIdAndIsFinishedTrue(String userId);
    
    /**
     * 统计练习的完成人数
     */
    long countByExerciseIdAndIsFinishedTrue(String exerciseId);
}